import os
import random
import re
import time

import requests
from bs4 import BeautifulSoup

user_agent = [
    "Mozilla/5.0 (Macintosh; U; Intel Mac OS X 10_6_8; en-us) AppleWebKit/534.50 (KHTML, like Gecko) Version/5.1 Safari/534.50",
    "Mozilla/5.0 (Windows; U; Windows NT 6.1; en-us) AppleWebKit/534.50 (KHTML, like Gecko) Version/5.1 Safari/534.50",
    "Mozilla/5.0 (Windows NT 10.0; WOW64; rv:38.0) Gecko/20100101 Firefox/38.0",
    "Mozilla/5.0 (Windows NT 10.0; WOW64; Trident/7.0; .NET4.0C; .NET4.0E; .NET CLR 2.0.50727; .NET CLR 3.0.30729; .NET CLR 3.5.30729; InfoPath.3; rv:11.0) like Gecko",
    "Mozilla/5.0 (compatible; MSIE 9.0; Windows NT 6.1; Trident/5.0)",
    "Mozilla/4.0 (compatible; MSIE 8.0; Windows NT 6.0; Trident/4.0)",
    "Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 6.0)",
    "Mozilla/4.0 (compatible; MSIE 6.0; Windows NT 5.1)",
    "Mozilla/5.0 (Macintosh; Intel Mac OS X 10.6; rv:2.0.1) Gecko/20100101 Firefox/4.0.1",
    "Mozilla/5.0 (Windows NT 6.1; rv:2.0.1) Gecko/20100101 Firefox/4.0.1",
    "Opera/9.80 (Macintosh; Intel Mac OS X 10.6.8; U; en) Presto/2.8.131 Version/11.11",
    "Opera/9.80 (Windows NT 6.1; U; en) Presto/2.8.131 Version/11.11",
    "Mozilla/5.0 (Macintosh; Intel Mac OS X 10_7_0) AppleWebKit/535.11 (KHTML, like Gecko) Chrome/17.0.963.56 Safari/535.11",
    "Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1; Maxthon 2.0)",
    "Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1; TencentTraveler 4.0)",
    "Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1)",
    "Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1; The World)",
    "Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1; Trident/4.0; SE 2.X MetaSr 1.0; SE 2.X MetaSr 1.0; .NET CLR 2.0.50727; SE 2.X MetaSr 1.0)",
    "Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1; 360SE)",
    "Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1; Avant Browser)",
    "Mozilla/4.0 (compatible; MSIE 7.0; Windows NT 5.1)",
    "Mozilla/5.0 (iPhone; U; CPU iPhone OS 4_3_3 like Mac OS X; en-us) AppleWebKit/533.17.9 (KHTML, like Gecko) Version/5.0.2 Mobile/8J2 Safari/6533.18.5",
    "Mozilla/5.0 (iPod; U; CPU iPhone OS 4_3_3 like Mac OS X; en-us) AppleWebKit/533.17.9 (KHTML, like Gecko) Version/5.0.2 Mobile/8J2 Safari/6533.18.5",
    "Mozilla/5.0 (iPad; U; CPU OS 4_3_3 like Mac OS X; en-us) AppleWebKit/533.17.9 (KHTML, like Gecko) Version/5.0.2 Mobile/8J2 Safari/6533.18.5",
    "Mozilla/5.0 (Linux; U; Android 2.3.7; en-us; Nexus One Build/FRF91) AppleWebKit/533.1 (KHTML, like Gecko) Version/4.0 Mobile Safari/533.1",
    "MQQBrowser/26 Mozilla/5.0 (Linux; U; Android 2.3.7; zh-cn; MB200 Build/GRJ22; CyanogenMod-7) AppleWebKit/533.1 (KHTML, like Gecko) Version/4.0 Mobile Safari/533.1",
    "Opera/9.80 (Android 2.3.4; Linux; Opera Mobi/build-1107180945; U; en-GB) Presto/2.8.149 Version/11.10",
    "Mozilla/5.0 (Linux; U; Android 3.0; en-us; Xoom Build/HRI39) AppleWebKit/534.13 (KHTML, like Gecko) Version/4.0 Safari/534.13",
    "Mozilla/5.0 (BlackBerry; U; BlackBerry 9800; en) AppleWebKit/534.1+ (KHTML, like Gecko) Version/6.0.0.337 Mobile Safari/534.1+",
    "Mozilla/5.0 (hp-tablet; Linux; hpwOS/3.0.0; U; en-US) AppleWebKit/534.6 (KHTML, like Gecko) wOSBrowser/233.70 Safari/534.6 TouchPad/1.0",
    "Mozilla/5.0 (SymbianOS/9.4; Series60/5.0 NokiaN97-1/20.0.019; Profile/MIDP-2.1 Configuration/CLDC-1.1) AppleWebKit/525 (KHTML, like Gecko) BrowserNG/7.1.18124",
    "Mozilla/5.0 (compatible; MSIE 9.0; Windows Phone OS 7.5; Trident/5.0; IEMobile/9.0; HTC; Titan)",
    "UCWEB7.0.2.37/28/999",
    "NOKIA5700/ UCWEB7.0.2.37/28/999",
    "Openwave/ UCWEB7.0.2.37/28/999",
    "Mozilla/4.0 (compatible; MSIE 6.0; ) Opera/UCWEB7.0.2.37/28/999",
    # iPhone 6：
    "Mozilla/6.0 (iPhone; CPU iPhone OS 8_0 like Mac OS X) AppleWebKit/536.26 (KHTML, like Gecko) Version/8.0 Mobile/10A5376e Safari/8536.25",

]

hd = {'User-Agent': random.choice(user_agent)}


# 封装方法用来爬取网页的源代码
def getHTML(url):
    try:
        r = requests.get(url, timeout=30, headers=hd)
        r.raise_for_status()  # 容错机制，若请求访问失败则返回的不是200，则返回字符串空
        r.encoding = r.apparent_encoding  # 设置编码方式，用解析返回网页源码得出的编码方式代替  UTF-8
        return r.text
    except:
        return ''


def get_image(year_no, web_no):
    count = 1
    page_no = 200
    web_no1 = ''
    leg_no = ''
    while count <= page_no:
        # 更改requests请求的头信息，用来防止网页的反爬虫，也可来伪装自己访问网页的信息，这里用最简单的chorm（浏览器）版本10代替
        if count != 1:
            web_no1 = str(web_no) + "_" + str(count)
        else:
            web_no1 = str(web_no)
        url = 'http://www.beautyleg6.com/siwameitui/' + str(year_no) + '/' + web_no1 + '.html'  # 爬取网页的url
        print('--------------------------------------------------------------------------------------------')
        print(url)
        urlInfo, leg_no = getImgUrlList(url, leg_no)
        # 保存图片，思路：将所有的图片保存在本地的一个文件夹下，用图片的url链接的后缀名来命名
        dir_name = 'D:\\like\\picture\\' + str(year_no)  # 设置文件夹的名字
        print(len(urlInfo))
        print(urlInfo)
        count1 = 0
        while len(urlInfo) == 0 and count1 <= 5:
            # time.sleep(random.random())
            urlInfo, leg_no = getImgUrlList(url, leg_no)
            print(len(urlInfo))
            print(urlInfo)
            count1 += 1
        if len(urlInfo) == 0:
            break
        if not os.path.exists(dir_name):  # os模块判断并创建
            os.mkdir(dir_name)
        if count == 1:
            with open(dir_name + '/' + 'url.txt', 'a') as uu:  # 保存每一个编码的首页地址
                uu.write(str(leg_no) + "\n")
                uu.write(url + '\n')
        for img_url in urlInfo:
            print(img_url)
            print(str(leg_no))
            picture_name = img_url.split('/')[-1]  # 提取图片url后缀
            time.sleep(random.random())
            reponse = requests.get(img_url)
            if not os.path.exists(dir_name + '/' + str(leg_no) + '-' + picture_name):
                with open(dir_name + '/' + str(leg_no) + '-' + picture_name, 'wb') as f:
                    f.write(reponse.content)
        count += 1

 
def getImgUrlList(url, leg_no):
    urlInfo = []  # 用来保存每一个图片拼接好的下载链接
    text = getHTML(url)
    soup = BeautifulSoup(text, 'html.parser')
    a = soup.find_all('a', string=re.compile('No.'), href=re.compile('.jpg'))
    for tag in a:
        new_url = tag.attrs['href']  # 得到img属性当中的src
        # print(tag.get_text())
        leg_no = tag.get_text()
        leg_no = str.replace(leg_no, '[', '')
        leg_no = str.replace(leg_no, '/', '')
        leg_no = str.replace(leg_no, ']', '').lstrip().rstrip()
        if 'http' in new_url and '.jpg' in new_url:
            urlInfo.append(new_url)  # 通过spilt函数剪切src并于之前的图片头部进行拼接，得到每个图片的下载地址
    return urlInfo, leg_no


if __name__ == "__main__":
    get_image('202308', 1489)
